查看原文
其他

漫画:什么是红黑树?(整合版)

The following article is from 程序员小灰 Author 小灰

前段时间,小灰发布了红黑树相关的文章,分成上下篇来讲解。


这一次,小灰把两篇文章做了整合,并且修正了红黑树删除部分的图片错误,感谢大家的指正。



—————  第二天  —————



————————————



二叉查找树(BST)具备什么特性呢?


1.子树上所有结点的值均小于或等于它的根结点的值。

2.子树上所有结点的值均大于或等于它的根结点的值。

3.左、右子树也分别为二叉排序树。


下图中这棵树,就是一颗典型的二叉查找树:





1.查看根结点9




2.根据二叉查找树左子树小、右子树大的特性,10 > 9,因此值为10的结点只可能在根结点的右子树当中,我们查看右孩子结点13




3.由于10 < 13,因此查看左孩子11




4.由于10 < 11,因此查看左孩子10,发现10正是要查找的结点:




假设初始的二叉查找树只有三个结点,根结点值为9,左孩子值为8,右孩子值为12:



接下来我们依次插入如下五个结点:7,6,5,4,3。依照二叉查找树的特性,结果会变成什么样呢?





1.结点是红色或黑色。

2.根结点是黑色。

3.每个叶子结点都是黑色的空结点(NIL结点)。

4 每个红色结点的两个子结点都是黑色。(从每个叶子到根的所有路径上不能有两个连续的红色结点)

5.从任一结点到其每个叶子的所有路径都包含相同数目的黑色结点。


下图中这棵树,就是一颗典型的红黑树:



什么情况下会破坏红黑树的规则,什么情况下不会破坏规则呢?我们举两个简单的例子:


1.向原红黑树插入值为14的新结点:




由于父结点15是黑色结点,因此这种情况并不会破坏红黑树的规则,无需做任何调整。



2.向原红黑树插入值为21的新结点:




由于父结点22是红色结点,因此这种情况打破了红黑树的规则4(每个红色结点的两个子结点都是黑色),必须进行调整,使之重新符合红黑树的规则。


变色:


为了重新符合红黑树的规则,尝试把红色结点变为黑色,或者把黑色结点变为红色。


下图所表示的是红黑树的一部分(子树),新插入的结点Y是红色结点,它的父亲结点X也是红色的,不符合规则4,因此我们可以把结点X从红色变成黑色:



但是,仅仅把一个结点变色,会导致相关路径凭空多出一个黑色结点,这样就打破了规则5。因此,我们需要对其他结点做进一步的调整,后文会详细说明。



左旋转:


逆时针旋转红黑树的两个结点,使得父结点被自己的右孩子取代,而自己成为自己的左孩子。说起来很怪异,大家看下图:



图中,身为右孩子的Y取代了X的位置,而X变成了自己的左孩子。此为左旋转。



右旋转:


顺时针旋转红黑树的两个结点,使得父结点被自己的左孩子取代,而自己成为自己的右孩子。大家看下图:



图中,身为左孩子的Y取代了X的位置,而X变成了自己的右孩子。此为右旋转。


局面1:新结点(A)位于树根,没有父结点。


(空心三角形代表结点下面的子树)


这种局面,直接让新结点变色为黑色,规则2得到满足。同时,黑色的根结点使得每条路径上的黑色结点数目都增加了1,所以并没有打破规则5。



局面2:新结点(B)的父结点是黑色。


这种局面,新插入的红色结点B并没有打破红黑树的规则,所以不需要做任何调整。



局面3:新结点(D)的父结点和叔叔结点都是红色。


这种局面,两个红色结点B和D连续,违反了规则4。因此我们先让结点B变为黑色:


这样一来,结点B所在路径凭空多了一个黑色结点,打破了规则5。因此我们让结点A变为红色:



这时候,结点A和C又成为了连续的红色结点,我们再让结点C变为黑色:



经过上面的调整,这一局部重新符合了红黑树的规则。



局面4:新结点(D)的父结点是红色,叔叔结点是黑色或者没有叔叔,且新结点是父结点的右孩子,父结点(B)是祖父结点的左孩子。


我们以结点B为轴,做一次左旋转,使得新结点D成为父结点,原来的父结点B成为D的左孩子:


这样一来,进入了局面5。


局面5:新结点(D)的父结点是红色,叔叔结点是黑色或者没有叔叔,且新结点是父结点的左孩子,父结点(B)是祖父结点的左孩子。


我们以结点A为轴,做一次右旋转,使得结点B成为祖父结点,结点A成为结点B的右孩子:


接下来,我们让结点B变为黑色,结点A变为红色:


经过上面的调整,这一局部重新符合了红黑树的规则。



以上就是红黑树插入操作所涉及的5种局面。


或许有人会问,如果局面4和局面5当中的父结点B是祖父结点A的右孩子该怎么办呢?


很简单,如果局面4中的父结点B是右孩子,则成为了局面5的镜像,原本的右旋操作改为左旋;如果局面5中的父结点B是右孩子,则成为了局面4的镜像,原本的左旋操作改为右旋。




给定下面这颗红黑树,新插入的结点是21:



显然,新结点21和它的父结点22是连续的红色结点,违背了规则4,我们应该如何调整呢?


让我们回顾一下刚才讲的5种局面,当前的情况符合局面3:

“新结点的父结点和叔叔结点都是红色。”


于是我们经过三次变色,22变为黑色,25变为红色,27变为黑色:



经过上面的调整,以结点25为根的子树符合了红黑树规则,但结点25和结点17成为了连续的红色结点,违背规则4。


于是,我们把结点25看做一个新结点,正好符合局面5的镜像:

“新结点的父结点是红色,叔叔结点是黑色或者没有叔叔,且新结点是父结点的右孩子,父结点是祖父结点的右孩子”


于是我们以根结点13为轴进行左旋转,使得结点17成为了新的根结点:



接下来,让结点17变为黑色,结点13变为红色:



如此一来,我们的红黑树变得重新符合规则。





二叉查找树是如何进行删除操作的呢?可以分成三种情况。


情况1,待删除的结点没有子结点:




上图中,待删除的结点12是叶子结点,没有孩子,因此直接删除即可:




情况2,待删除的结点有一个孩子:




上图中,待删除的结点13只有左孩子,于是我们让左孩子结点11取代被删除的结点,结点11以下的结点关系无需变动:




情况3,待删除的结点有两个孩子:



上图中,待删除的结点5有两个孩子,这种情况比较复杂。此时,我们需要选择与待删除结点最接近的结点来取代它。


上面的例子中,结点3仅小于结点5,结点6仅大于结点5,两者都是合适的选择。但习惯上我们选择仅大于待删除结点的结点,也就是结点6来取代它。


于是我们复制结点6到原来结点5的位置:



被选中的结点6,仅大于结点5,因此一定没有左孩子。所以我们按照情况1或情况2的方式,删除多余的结点6:




红黑树的特性(规则)如下:


1.结点是红色或黑色。

2.根结点是黑色。

3.每个叶子结点都是黑色的空结点(NIL结点)。

4.每个红色结点的两个子结点都是黑色。(从每个叶子到根的所有路径上不能有两个连续的红色结点)

5.从任一结点到其每个叶子的所有路径都包含相同数目的黑色结点。



下面我们通过一个例子,来看一看删除红黑树的结点会对规则产生怎样的影响:




上图的这颗红黑树,待删除的是黑色结点1,有一个右孩子。根据二叉查找树的删除流程,我们让右孩子结点6直接取代结点1:



显然,这颗新的二叉树打破了两个规则:


规则4. 每个红色结点的两个子结点都是黑色。

规则5. 从任一结点到其每个叶子的所有路径都包含相同数目的黑色结点。



第一步:如果待删除结点有两个非空的孩子结点,转化成待删除结点只有一个孩子(或没有孩子)的情况。



上面例子是一颗红黑树的局部,标数字的三角形代表任意形态的子树,假设结点8是待删除结点。


根据上文讲解的二叉查找树删除流程,由于结点8有两个孩子,我们选择仅大于8的结点10复制到8的位置,结点颜色变成待删除结点的颜色:


接下来我们需要删除红色的结点10:



红色结点10能成为仅大于8的结点,必定没有左孩子结点,所以问题转换成了待删除结点只有一个右孩子(或没有孩子)的情况。接下来我们进入第二步。



第二步:根据待删除结点和其唯一子结点的颜色,分情况处理。


情况1,自身是红色,子结点是黑色:



这种情况最简单,按照二叉查找树的删除操作,删除结点1即可:



情况2,自身是黑色,子结点是红色:


这种情况也很简单,首先按照二叉查找树的删除操作,删除结点1:


此时,这条路径凭空减少了一个黑色结点,那么我们把结点2变成黑色即可:



情况3,自身是黑色,子结点也是黑色,或者子结点是空叶子结点:



这种情况最复杂,涉及到很多变化。首先我们还是按照二叉查找树的删除操作,删除结点1:



显然,这条路径上减少了一个黑色结点,而且结点2再怎么变色也解决不了。


这时候我们进入第三步,专门解决父子双黑的情况。



第三步:遇到双黑结点,在子结点顶替父结点之后,分成6种子情况处理。


子情况1,结点2是红黑树的根结点:


此时所有路径都减少了一个黑色结点,并未打破规则,不需要调整。


子情况2,结点2的父亲、兄弟、侄子结点都是黑色:


此时,我们直接把结点2的兄弟结点B改为红色:



这样一来,原本结点2所在的路径少了一个黑色结点,现在结点B所在的路径也少了一个黑色结点,两边“扯平”了。


可是,结点A以下的每一条路径都减少了一个黑色结点,与结点A之外的其他路径又造成了新的不平衡啊?


没关系,我们让结点A扮演原先结点2的角色,进行递归操作,重新判断各种情况。


子情况3,结点2的兄弟结点是红色:


首先以结点2的父结点A为轴,进行左旋:


然后结点A变成红色、结点B变成黑色:



这样的意义是什么呢?结点2所在的路径仍然少一个黑色结点呀?


别急,这样的变化有可能转换成子情况4、5、6中的任意一种,在子情况4、5、6当中会进一步解决。


子情况4,结点2的父结点是红色,兄弟和侄子结点是黑色:




这种情况,我们直接让结点2的父结点A变成黑色,兄弟结点B变成红色:



这样一来,结点2的路径补充了黑色结点,而结点B的路径并没有减少黑色结点,重新符合了红黑树的规则。


子情况5,结点2的父结点随意,兄弟结点B是黑色右孩子,左侄子结点是红色,右侄子结点是黑色:



这种情况下,首先以结点2的兄弟结点B为轴进行右旋:


接下来结点B变为红色,结点C变为黑色:


这样的变化转换成了子情况6。


子情况6,结点2的父结点随意,兄弟结点B是黑色右孩子,右侄子结点是红色:




首先以结点2的父结点A为轴左旋:



接下来让结点A和结点B的颜色交换,并且结点D变为黑色:



这样是否解决了问题呢?


经过结点2的路径由(随意+黑)变成了(随意+黑+黑),补充了一个黑色结点;


经过结点D的路径由(随意+黑+红)变成了(随意+黑),黑色结点并没有减少。


所以,这时候重新符合了红黑树的规则。



以上就是红黑树删除的全过程。


给定下面这颗红黑树,待删除的是结点17:



第一步,由于结点17有两个孩子,子树当中仅大于17的结点是25,所以把结点25复制到17位置,保持黑色:




接下来,我们需要删除原本的结点25:




这个情况正好对应于第二步的情况三,即待删除结点是黑色,子结点是空叶子结点。


于是我们删除框框中结点25,进入第三步:



此时,框框中的结点虽然是空叶子结点,但仍然可以用于判断局面,当前局面符合子情况5的镜像:





于是我们通过左旋和变色,把子树转换成情况6的镜像:



再经过右旋、变色,子树最终成为了下面的样子:






这样一来,整颗二叉树又重新符合了红黑树的规则。






往期推荐

万字详解|手撕 9大排序算法!

2020-09-11

URL 去重的 6 种方案!(附详细代码)

2020-09-09

字符串操作的12个小技巧!

2020-09-10


关注下方二维码,收获更多干货!

    您可能也对以下帖子感兴趣

    文章有问题?点此查看未经处理的缓存